package com.xzmzhp.config;


import com.xzmzhp.security.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.http.HttpMethod;
import org.springframework.security.authentication.AuthenticationManager;
import org.springframework.security.config.annotation.ObjectPostProcessor;
import org.springframework.security.config.annotation.authentication.builders.AuthenticationManagerBuilder;
import org.springframework.security.config.annotation.method.configuration.EnableGlobalMethodSecurity;
import org.springframework.security.config.annotation.web.builders.HttpSecurity;
import org.springframework.security.config.annotation.web.configuration.WebSecurityConfigurerAdapter;
import org.springframework.security.config.http.SessionCreationPolicy;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.security.web.access.intercept.FilterSecurityInterceptor;
import org.springframework.security.web.authentication.UsernamePasswordAuthenticationFilter;
import org.springframework.security.web.authentication.logout.LogoutFilter;
import org.springframework.web.filter.CorsFilter;

/**
 * @author 王源(xm)
 * @Date: 2020/11/15 15:24
 * @Description: Security配置类
 */
@EnableGlobalMethodSecurity(prePostEnabled = true, securedEnabled = true)
public class SecurityConfig extends WebSecurityConfigurerAdapter {

    /**
     * 登录逻辑处理类
     */
    @Qualifier("userDetailsService")
    @Autowired
    private MyUserDetailsService userDetailsService;

    /**
     * 登出处理类
     */
    @Autowired
    private MyLogoutHandler logoutHandler;


    /**
     * 登陆失败处理类
     */
    @Autowired
    private LoginFailureHandler loginFailureHandler;

    /**
     * 跨域过滤器
     */
    @Autowired
    private CorsFilter corsFilter;

    /**
     * token过滤
     */
    @Autowired
    private TokenAuthenticationFilter tokenAuthenticationFilter;

    /**
     * 获取当前请求所需要的权限
     */
    @Autowired
    private FilterInvocationSecurity filter;

    /**
     * 权限决策器
     */
    @Autowired
    private RoleAccessDecisionManager roleAccessDecisionManager;


    /**
     * token工具类
     */
    @Autowired
    private TokenManager tokenManager;

    /**
     * 解决 无法直接注入 AuthenticationManager
     *
     * @return AuthenticationManager对象
     * @throws Exception 全局错误
     */
    @Bean
    @Override
    public AuthenticationManager authenticationManagerBean() throws Exception {
        return super.authenticationManagerBean();
    }

    /**
     * 基础认证与权限配置
     *
     * @param httpSecurity 请求
     * @throws Exception 全局错误
     *                   anyRequest          |   匹配所有请求路径
     *                   access              |   SpringEl表达式结果为true时可以访问
     *                   anonymous           |   匿名可以访问
     *                   denyAll             |   用户不能访问
     *                   fullyAuthenticated  |   用户完全认证可以访问（非remember-me下自动登录）
     *                   hasAnyAuthority     |   如果有参数，参数表示权限，则其中任何一个权限可以访问
     *                   hasAnyRole          |   如果有参数，参数表示角色，则其中任何一个角色可以访问
     *                   hasAuthority        |   如果有参数，参数表示权限，则其权限可以访问
     *                   hasIpAddress        |   如果有参数，参数表示IP地址，如果用户IP和参数匹配，则可以访问
     *                   hasRole             |   如果有参数，参数表示角色，则其角色可以访问
     *                   permitAll           |   用户可以任意访问
     *                   rememberMe          |   允许通过remember-me登录的用户访问
     *                   authenticated       |   用户登录后可访问
     */
    @Override
    protected void configure(HttpSecurity httpSecurity) throws Exception {
        httpSecurity
                // CSRF禁用，因为不使用session
                .csrf().disable()
                // 认证失败处理类
                .exceptionHandling().authenticationEntryPoint(loginFailureHandler).and()
                // 基于token，所以不需要session
                .sessionManagement().sessionCreationPolicy(SessionCreationPolicy.STATELESS).and()
                // 过滤请求
                .authorizeRequests()
                // 对于登录login 验证码captchaImage 允许匿名访问
                .antMatchers("/login").anonymous()
                .antMatchers(
                        HttpMethod.GET,
                        "/*.html",
                        "/**/*.html",
                        "/**/*.css",
                        "/**/*.js"
                ).permitAll()
                .antMatchers("/menu/**").anonymous()
                .withObjectPostProcessor(new ObjectPostProcessor<FilterSecurityInterceptor>() {
                    @Override
                    public <O extends FilterSecurityInterceptor> O postProcess(O o) {
                        o.setAccessDecisionManager(roleAccessDecisionManager);
                        o.setSecurityMetadataSource(filter);
                        return o;
                    }
                })
                // 除上面外的所有请求全部需要鉴权认证
                .anyRequest().authenticated()
                .and()
                .headers().frameOptions().disable();
        httpSecurity.logout().logoutUrl("/logout").logoutSuccessHandler(logoutHandler);
        // 添加JWT filter
        httpSecurity.addFilterBefore(tokenAuthenticationFilter, UsernamePasswordAuthenticationFilter.class);
        // 添加CORS filter
        httpSecurity.addFilterBefore(corsFilter, TokenAuthenticationFilter.class);
        httpSecurity.addFilterBefore(corsFilter, LogoutFilter.class);
    }

    /**
     * 将BCryptPasswordEncoder注入到spring容器
     */
    @Bean
    public BCryptPasswordEncoder bCryptPasswordEncoder() {
        return new BCryptPasswordEncoder();
    }

    /**
     * 身份认证接口
     */
    @Override
    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
        auth.userDetailsService(userDetailsService).passwordEncoder(bCryptPasswordEncoder());
    }
}
